apply method in python